import os
from cognlp.utils.vocabulary import Vocabulary


class DataModule:
    def __init__(self, path):
        if os.path.exists(os.path.join(path, "vocabulary.txt")):
            self.vocabulary = Vocabulary.load(os.path.join(path, "vocabulary.txt"))
